#pragma once
#include <iostream>
#include <string>
#include "Log.hpp"
#include "Thread.hpp"
#include <vector>
#include <queue>
#include "Cond.hpp"
#include "Mutex.hpp"

namespace ThreadPoolModule
{
    using namespace ThreadModlue;
    using namespace LogModule;
    using namespace MutexModule;
    using namespace CondModule;
    static const int gnum = 5;

    template <typename T>
    class ThreadPool
    {
    private:
        void WakeupAllThread()
        {
            // 如果有休眠的线程，才能全部唤醒
            if (sleepnum)
                _cond.Broadcast(); // 在条件变量下唤醒
            LOG(LogLevel::INFO) << "唤醒所有线程";
        }

        void WakeupOne()
        {
            _cond.Signal();
            LOG(LogLevel::INFO) << "唤醒一个休眠线程";
        }

        ThreadPool(int num = gnum) : _num(num), _isrunning(false), sleepnum(0)
        {
            for (int i = 0; i < num; i++)
            {
                _threads.emplace_back(
                    [this]()
                    {
                        HandlerTask();
                    });
            }
        }

        // 启动线程
        void Start()
        {
            // 如果线程池已经启动了，那就不用再启动了
            if (_isrunning == true)
                return;
            _isrunning = true;

            for (auto &thread : _threads)
            {
                thread.Start();
                LOG(LogLevel::INFO) << "创建线程成功" << thread.Name();
            }
        }

        // 将拷贝构造和赋值设为0，禁用
        ThreadPool(const ThreadPool<T> &) = delete;
        ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;

    public:
        // 获取单例指针
        // static 就能不用对象调用就可以调用,但不能调用成员
        static ThreadPool<T> *GetInstance()
        {
            if (inc == nullptr)
            {
                //因为可能有多线程来申请单例，所以有两层if
                LockGuard lockguard(_lock); // 锁
                LOG(LogLevel::DEBUG) << "获取单例...";
                if (inc == nullptr)
                {
                    LOG(LogLevel::DEBUG) << "首次使用单例，创建之...";
                    inc = new ThreadPool<T>();
                    inc->Start();
                }
                return inc;
            }
        }
        void Stop()
        {
            if (_isrunning == false)
                return;
            _isrunning = false;

            // 停止线程池，就要停止所有线程，所以我们要把所有线程唤醒
            WakeupAllThread();
        }

        void Join()
        {
            for (auto &thread : _threads)
            {
                thread.Join();
            }
        }
        // 启动线程执行任务，任务就是打印日志
        void HandlerTask()
        {
            char name[128];
            pthread_getname_np(pthread_self(), name, sizeof(name));
            while (true)
            {
                T t;
                {
                    LockGuard lockguard(_mutex);
                    // 线程队列为空&&线程池没有退出
                    while (_taskq.empty() && _isrunning)
                    {
                        // 休眠
                        sleepnum++;
                        _cond.Wait(_mutex);
                        // 唤醒
                        sleepnum--;
                    }

                    // 线程被唤醒了
                    if (!_isrunning && _taskq.empty())
                    {
                        LOG(LogLevel::INFO) << name << "退出了，线程退出&&任务队列为空";
                        break;
                    }
                    // 一定有任务
                    t = _taskq.front();
                    _taskq.pop();
                }
                // 处理任务，因为可以同时处理任务，不用加锁
                t();
            }
        }

        // 向线程队列里面输入任务
        bool Enqueue(const T &in)
        {
            if (_isrunning)
            {
                LockGuard lockguard(_mutex);
                _taskq.push(in);
                // 线程中所有线程都在休眠，那就叫醒一个去干活
                if (_threads.size() - sleepnum == 0)
                {
                    WakeupOne();
                }
                return true;
            }
            return false;
        }

        ~ThreadPool()
        {
        }

    private:
        int _num;
        std::vector<Thread> _threads;
        std::queue<T> _taskq; // 任务队列
        Cond _cond;           // 条件变量
        Mutex _mutex;         // 锁

        bool _isrunning;
        int sleepnum;

        // 单例指针
        static ThreadPool<T> *inc;
        static Mutex _lock;
    };

    // 单例
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::inc = nullptr; // 指针初始化

    // 锁
    template <typename T>
    Mutex ThreadPool<T>::_lock;
}
